package znet

import (
	"bytes"
	"encoding/binary"
	"errors"
	ziface "zinx/Ziface"
	"zinx/utlis"
)

/*
提供TCP的封包拆包的方法
自定义的应用层协议
*/
type DataPack struct{}

func NewDataPack() *DataPack {
	return &DataPack{}
}

// 获取包头长度：DataLen+Id
func (data *DataPack) GetHeadLen() uint32 {
	//Id uint32(4字节) +  DataLen uint32(4字节)
	return 8
}

// 封包方法
func (data *DataPack) Pack(msg ziface.IMessage) ([]byte, error) {

	//创建一个存放bytes字节的缓冲
	databuf := bytes.NewBuffer([]byte{})

	//写入DataLen,binary,LittleEndian表示按照小字节序写入
	if err := binary.Write(databuf, binary.LittleEndian, msg.GetDataLen()); err != nil {
		return nil, err
	}
	//写入Msgid
	if err := binary.Write(databuf, binary.LittleEndian, msg.GetMsgId()); err != nil {
		return nil, err
	}
	//写入Data
	if err := binary.Write(databuf, binary.LittleEndian, msg.GetData()); err != nil {
		return nil, err
	}
	return databuf.Bytes(), nil
}

// 解包方法
func (data *DataPack) UnPack(buf []byte) (ziface.IMessage, error) {

	//创建一个从输入二进制数据的ioReader
	dataBuff := bytes.NewReader(buf)
	//创建message
	msg := &Message{}
	//读取DataLen
	if err := binary.Read(dataBuff, binary.LittleEndian, &msg.DataLen); err != nil {
		return nil, err
	}
	//读取id
	if err := binary.Read(dataBuff, binary.LittleEndian, &msg.Id); err != nil {
		return nil, err
	}

	
	//判断dataLen的长度是否超出我们允许的最大包长
	if utlis.GlovbalObject.MaxPackageSize > 0 && msg.DataLen > uint32(utlis.GlovbalObject.MaxPackageSize) {
		return nil, errors.New("Too large msg data")
	}

	//这里只需要把head的数据拆包出来就可以了，然后再通过head的长度，再从conn读取一次数据,不用专门去读数据
	return msg, nil
}
